package socket;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

public class Server {
    private ServerSocket serverSocket;
    private List<PrintWriter> allOut = new ArrayList<>();

    public Server() {
        ServerSocket server = null;
        try {
            System.out.println("正在启动服务端......");
            serverSocket = new ServerSocket(8088);
            System.out.println("服务端启动完毕！");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void start() {
        try {
            while (true) {
                System.out.println("等待客户端连接......");
                Socket socket = serverSocket.accept();
                System.out.println("客户端连接成功！");
                ClientHandler handler = new ClientHandler(socket);
                Thread thread = new Thread(handler);
                thread.start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) throws IOException {
        Server server = new Server();
        server.start();
    }

    private class ClientHandler implements Runnable {
        private Socket socket;
        private String host;

        public ClientHandler(Socket socket) {
            this.socket = socket;
            host = socket.getInetAddress().getHostAddress();
        }

        public void run() {
            PrintWriter pw = null;
            try {
                InputStream in = socket.getInputStream();
                InputStreamReader isr = new InputStreamReader(in, StandardCharsets.UTF_8);
                BufferedReader br = new BufferedReader(isr);
                OutputStream out = socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out, StandardCharsets.UTF_8);
                BufferedWriter bw = new BufferedWriter(osw);
                pw = new PrintWriter(bw, true);
                synchronized (allOut) {
                    allOut.add(pw);
                }
                System.out.println(host + "已上线，当前在线人数：" + allOut.size());

                String mes;
                while (((mes = br.readLine()) != null) && !mes.equals("exit")) {
                    System.out.println("客户端" + host + "说：" + mes);
                    synchronized (allOut) {
                        for (PrintWriter p : allOut) {
                            p.println("客户端" + host + "说：" + mes);
                        }
                    }
                }
                System.out.println(host + "已下线，当前在线人数：" + (allOut.size()-1));
            } catch (IOException e) {
                System.out.println(host + "已下线，当前在线人数：" + (allOut.size()-1));
            } finally {
                synchronized (allOut) {
                    allOut.remove(pw);
                }

            }
        }
    }
}
